package inits

import (
	"fmt"
	"github.com/go-redis/redis"
	"github.com/spf13/viper"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"time"
	"user-server/global"
)

func Inits() {
	InitConf()
	InitMysql()
	InitRedis()
}

func InitConf() {
	viper.SetConfigFile("./dev.yaml")
	err := viper.ReadInConfig()
	if err != nil {
		fmt.Println("配置文件读取失败")
	}
	err = viper.Unmarshal(&global.Conf)
	if err != nil {
		fmt.Println("配置文件解析失败")
	}
	fmt.Println(&global.Conf)
}

func InitMysql() {
	var err error
	var conf = global.Conf.Mysql
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8mb4&parseTime=True&loc=Local", conf.User, conf.Password, conf.Host, conf.Port, conf.Database)
	global.DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		fmt.Println("数据库连接失败")
		return
	}
	fmt.Println("数据库连接成功")
	// 获取通用数据库对象 sql.DB ，然后使用其提供的功能
	sqlDB, err := global.DB.DB()

	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(10)

	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)

	// SetConnMaxLifetime 设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)
}

func InitRedis() {
	global.Rdb = redis.NewClient(&redis.Options{
		Addr:     global.Conf.Redis.Addr,
		Password: global.Conf.Redis.Password, // no password set
		DB:       global.Conf.Redis.DB,       // use default DB
	})
}
